#include <iostream>
#include "UdpServer.hpp"

int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        std::cerr << "Usage:" << argv[0] << "localport" << std::endl;
        DIE(USAGE_ERR);
    }
    uint16_t port = std::stoi(argv[1]);

    LogModule::ENABLE_CONSOLE_LOG();

    // 用户管理模块
    std::shared_ptr<UserManager> um_ptr = std::make_shared<UserManager>();
    // 网络模块
    std::unique_ptr<UdpServer> svr_uptr = std::make_unique<UdpServer>(port);
    svr_uptr->InitServer();
    svr_uptr->RegisterService([&um_ptr](InetAddr &id)
                              { um_ptr->AddUser(id); }, 
                              [&um_ptr](int sockfd, const std::string &message)
                              { um_ptr->Router(sockfd, message); },
                              [&um_ptr](InetAddr& id){um_ptr->DelUser(id);}
                            );
    svr_uptr->Start();
    return 0;
}